<?php

tsload(APPS_PATH.'/admin/Lib/Action/AdministratorAction.class.php');

/**
 * 升级程序.
 *
 * @author Medz Seven <lovevipdsw@vip.qq.com>
 **/
class UpgradeAction extends AdministratorAction
{
    /**
     * 执行前.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function _initialize()
    {
        extension_loaded('zlib') or $this->error('服务器未安装php的zlib拓展，无法使用在线升级功能');
        function_exists('gzcompress') or $this->error('服务器不支持gzcompress函数，无法使用在线升级功能');
        parent::_initialize();
    }

    /**
     * 后台检测.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function ajxjCheck()
    {
        ob_end_clean();
        ob_start();
        header('Content-Type: application/json; charset=utf-8');

        echo file_get_contents(C('UPURL').'?v='.C('VERSION'));

        ob_end_flush();
        exit;
    }

    /**
     * 检查是否有更新.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function check()
    {
        if (version_compare(PHP_VERSION, '5.3.12', '<')) {
            $this->showError('您当前PHP: '.PHP_VERSION.'版本低于ThinkSNS4运行的最低版本PHP:5.3.12');
        }

        $url = C('UPURL').'?v='.C('VERSION');
        $data = file_get_contents($url);

        $data or $this->showError('您的服务器无法从升级服务器获取升级数据！');

        // 保存线上包详情
        $filename = DATA_PATH.'/'.'upgrade/upgrade.json';
        file_put_contents($filename, $data);

        $data = json_decode($data, false);

        function_exists('json_decode') or $this->showError('你的服务器不支持json_decode函数');

        switch ($data->status) {
            case 1:
                $this->showSuccess('', '暂时没有更新');
                break;

            case 2:
                $this->showUpgrade($data->message, $data->url);
                break;

            case 0:
            default:
                $this->showError($data->message, '无法获得更新');
                break;
        }
        unset($url, $data);
    }

    /**
     * 显示消息.
     *
     * @param string $message 消息
     * @param string $type    [success|error] 消息类型
     * @param string $url     跳转的url
     * @param int    $s       等待的时间
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function showMessage($message, $type, $url = false, $s = 3)
    {
        $this->assign('message', $message);
        $this->assign('type', $type);
        $this->assign('url', $url);
        $this->assign('s', intval($s));
        $this->display('message');
        exit;
    }

    /**
     * 显示正确消息.
     *
     * @param string $message        消息
     * @param string $defaultMessage 默认消息
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    private function showSuccess($message = '', $defaultMessage = '正确')
    {
        $message or $message = $defaultMessage;
        $this->showMessage($message, 'success', false);
    }

    /**
     * 显示错误消息.
     *
     * @param string $message        消息
     * @param string $defaultMessage 默认消息
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    private function showError($message = '', $defaultMessage = '错误')
    {
        $message or $message = $defaultMessage;
        $this->showMessage($message, 'error', false);
    }

    /**
     * 显示升级信息.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    private function showUpgrade($log, $downUrl)
    {
        $this->savePostUrl = U('admin/Upgrade/step1', array('upurl' => urlencode($downUrl)));
        $this->pageTitle['showUpgrade'] = '更新日志';
        $this->submitAlias = '立即升级';
        $this->pageKeyList = array('log', 'tips');
        $this->opt['tips'] = '<pre style="color:#f00;font-size:20px">
=====如果在线升级：<font color="#000">
1.升级前请做好网站程序和数据备份。
2.如果您网站经过修改，请勿使用在线升级。
3.升级需要ThinkSNS程序文件和目录拥有可写，可读权限，升级前先确定
4.升级前一定要做好数据备份。
5.因各种因素，您无法使用在线升级，那么，请 <a href="'.$downUrl.'">点击这里</a>手动下载升级包进行手动升级。
</font>
=====如果手动升级：<font color="#000">
1.点击上方链接，下载补丁。
2.解压下载的补丁。
3.将补丁覆盖到您的ThinkSNS目录下。
4.覆盖完成，<a href="'.U('admin/Upgrade/step3').'">点击这里</a>完成最后的升级工作。</font>
PS：手动升级覆盖文件后千万不要刷新本页面，直接点击上方最后升级步骤。否则数据库有缺失自行负责。
<pre/>';
        $this->onsubmit = 'confirm(\'确定要升级吗？\')';
        $this->displayConfig(array(
            'log'  => $log,
            'tips' => $this->opt['tips'],
        ));
    }

    /**
     * 升级程序第一步 下载增量包.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function step1()
    {
        $downUrl = $_GET['upurl'];
         $downUrl = urldecode($downUrl); if (preg_match("#^http://demo\.thinksns\.com#sim", $downUrl) == 0) { die("request error!"); } 
        $path = DATA_PATH.'/'.'upgrade/'.basename($downUrl);

        // # 备份老配置文件
        $oldConf = file_get_contents(CONF_PATH.'/thinksns.conf.php');
        file_put_contents(DATA_PATH.'/old.thinksns.conf.php', $oldConf);

        // # 下载增量包
        is_dir(dirname($path)) or mkdir(dirname($path), 0777, true);
        file_put_contents($path, file_get_contents($downUrl));
        file_exists($path) or $this->showError('下载升级包失败，请检查'.dirname($path).'目录是否可写，如果可写，请刷新重试！');

        // 验证hash判断包是否合法。
        $filename = dirname($path).'/upgrade.json';
        $data = file_get_contents($filename);
        $data = json_decode($data, false);
        if (md5_file($path) != $data->md5) {
            $this->showError('更新包校验失败，请重新执行升级.');
        }

        $sqlPath = dirname($path).'/'.'upgrade.sql';
        $delFile = dirname($path).'/'.'deleteFiles.php';

        file_exists($delFile) and file_put_contents($delFile, '<?php return array(); ?>');
        file_exists($sqlPath) and file_put_contents($sqlPath, '-- 暂无升级 SQL --');

        // # 解压增量包
        $zip = new MedzZip();
        $zip->init() or $this->showError('初始化解压程序失败！');

        $list = $zip->extract($path);

        foreach ($list as $info) {
            $filename = SITE_PATH.'/'.$info['filename'];
            if (!file_exists($filename)) {
                is_dir(dirname($filename)) or mkdir(dirname($filename), 0777, true);
                is_dir(dirname($filename)) or $this->showError('目录'.dirname($filename).'创建失败，请赋予0755权限');
                file_put_contents($filename, $info['data']);
                file_exists($filename) or $this->showError($filename.'文件写入失败，请赋予'.dirname($filename).'目录0755权限');
            } elseif (!is_writable($filename)) {
                $this->showError($filename.'文件写入失败，请确认该文件为可写状态');
            }
        }

        $this->showMessage('权限检查成功，程序自动进入下一步（请勿操作页面）', 'success', U('admin/Upgrade/step2', array(
            'filename' => urlencode(basename($path)),
        )), 3);
    }

    /**
     * 升级程序第二步 - 执行文件替换.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function step2()
    {
        $filename = $_GET['filename'];
        $filename = urldecode($filename);
        $filename = DATA_PATH.'/'.'upgrade/'.$filename;

        $zip = new MedzZip();
        $zip->init();

        foreach ($zip->extract($filename) as $info) {
            $path = $info['filename'];
            $data = $info['data'];

            unset($info);

            $path = SITE_PATH.'/'.$path;
            $dir = dirname($path);

            is_dir($dir) or mkdir($dir, 0777, true);
            file_put_contents($path, $data);
            unset($data);
        }

        $this->showMessage('文件升级完成，程序自动进入下一步（请勿操作页面）', 'success', U('admin/Upgrade/step3'));
    }

    /**
     * 升级第三步 - 删除升级标记需要删除的文件 和执行sql文件.
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    public function step3()
    {
        $sqlFilePath = DATA_PATH.'/upgrade/upgrade.sql';
        $delFile = dirname($sqlFilePath).'/deleteFiles.php';

        // # 删除废弃文件
        if (file_exists($delFile)) {
            $delFile = include $delFile;
            foreach ($delFile as $filename) {
                $filename = SITE_PATH.'/'.$filename;
                $this->rm($filename, true);
            }
        }

        // # 执行sql
        if (file_exists($sqlFilePath)) {
            $result = D('')->executeSqlFile($sqlFilePath);
            if (isset($result['error_code'])) {
                // # 回滚配置文件
                $oldConf = file_get_contents(DATA_PATH.'/old.thinksns.conf.php');
                file_put_contents(CONF_PATH.'/thinksns.conf.php', $oldConf);

                $this->showMessage($result['error_code'].',请重新执行升级', 'error', U('admin/upgrade/check'));
            }
        }
        A('Tool', 'Admin')->cleancache();
        ob_end_clean();
        $this->showMessage('升级成功', 'success', U('admin/upgrade/check'));
    }

    /**
     * 删除函数，自带递归.
     *
     * @param string $path      删除的地址
     * @param bool   $recursive 是否递归 [true]
     *
     * @return bool
     *
     * @author Medz Seven <lovevipdsw@vip.qq.com>
     **/
    protected function rm($path, $recursive = true)
    {
        // # 判断是否是文件
        if (is_file($path)) {
            return unlink($path);

        // # 判断是否是目录，判断是否递归，不递归，直接尝试删除
        } elseif (is_dir($path) and !$recursive) {
            return rmdir($path);

        // # 判断是否是目录，如果是目录，则递归删除
        } elseif (is_dir($path) && file_exists($path)) {
            // # 打开目录资源
            $handle = opendir($path);

            // # 单条读取内容
            while (false !== ($file = readdir($handle))) {
                // # 判断内容是否是 .|.. 如果是，跳过
                if ($file == '.' or $file == '..') {
                    // # 跳过当前循环
                    continue;
                }

                // # 递归执行自己
                $this->rm($path.'/'.$file, $recursive);
            }

            // # 关闭目录资源
            closedir($handle);

            // # 尝试删除当前目录
            return rmdir($path);
        }

        // # 如果以上都不满足，则返回真
        return true;
    }
} // END class UpgradeAction extends AdministratorAction
